package com.atguigu.search;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @author Amos
 * @date 2022/6/16
 */

public class BinarySearch {
    public static void main(String[] args) {
        int arr[] = {1, 8, 10, 89, 1000, 1000, 1234};
//        System.out.println(arr);

        List<Integer> resIndex = binarySearch2(arr, 0, arr.length - 1, 1000);
        System.out.println("resIndex=" + resIndex);
    }

    // 二分查找算法

    /**
     * @param arr       数组
     * @param left      左边的索引
     * @param right     右边的索引
     * @param findValue 要查找的值
     * @return 如果找到就返回下标
     */
    public static int binarySearch(int[] arr, int left, int right, int findValue) {

        // 当left > right 时，说明递归整个数组，但是没有找到
        if (left > right) {
            return -1;
        }
        int mid = (left + right) / 2;
        int midVal = arr[mid];

        if (findValue > midVal) { // 向右递归
            return binarySearch(arr, mid + 1, right, findValue);
        } else if (findValue < midVal) {
            return binarySearch(arr, left, mid - 1, findValue);
        } else {
            return mid;
        }

    }

    // 完成一个课后思考题
    /*
        课后思考题：{1,8,10,89,1000,1000,1234}当一个有序数组中，
        有多个相同的数值时，如何将所有的数值都查找到，比如这里的1000

        思路分析：
        1. 在找到mid索引时，不要马上返回
        2. 向mid索引值的左边扫描，将所有满足1000，的元素的下标，加入到集合ArrayList
        3. 向mid索引值的右边扫描，将所有满足1000，的元素的下标，加入到集合ArrayList
        4. 将ArrayList返回
     */
    public static List<Integer> binarySearch2(int[] arr, int left, int right, int findVal) {
        // 当left > right时，说明递归这个数组，但是没有找到
        if (left > right) {
            return new ArrayList<Integer>();
        }

        int mid = (left + right) / 2;
        int midValue = arr[mid];

        if (findVal > midValue) { // 向右递归
            return binarySearch2(arr, mid + 1, right, findVal);
        } else if (findVal < midValue) { // 向左递归
            return binarySearch2(arr, left, mid - 1, findVal);
        } else {
            List<Integer> resIndexlist = new ArrayList<>();

            // 向mid索引值的左边扫描，将所有满足1000， 的元素的下标，加入到List中
            int temp = mid - 1;
            while (true) {
                if (temp < 0 || arr[temp] != findVal) { // 退出
                    break;
                }
                // 否则，就将temp放入到resIndexlist中
                resIndexlist.add(temp);
                temp -= 1; // temp左移
            }
            resIndexlist.add(mid);

            // 向mid索引值的右边扫描，将所有满足1000， 的元素的下标，加入到List中
            temp = mid + 1;
            while (true) {
                if (temp > arr.length - 1 || arr[temp] != findVal) {
                    break;
                }
                // 否则，就将temp放入到resIndexlist中
                resIndexlist.add(temp);
                temp += 1; // temp右移
            }

            return resIndexlist;
        }

    }
}
